# Copyright (c) Huawei Technologies Co., Ltd. 2020. All rights reserved.

# CMake lowest version requirement
cmake_minimum_required(VERSION 3.5.1)

# project information
project(acl_execute_add)

# Compile options
add_compile_options(-std=c++11)

set(CMAKE_RUNTIME_OUTPUT_DIRECTORY "../output")
set(CMAKE_LIBRARY_OUTPUT_DIRECTORY "../output")

set(INC_PATH $ENV{DDK_PATH})

if (NOT DEFINED ENV{DDK_PATH})
    set(INC_PATH "/usr/local/Ascend/ascend-toolkit/latest")
    message(STATUS "set default INC_PATH: ${INC_PATH}")
else ()
    message(STATUS "env INC_PATH: ${INC_PATH}")
endif()

set(CUST_PKG_PATH "${INC_PATH}/opp/vendors/customize/op_api")

set(LIB_PATH $ENV{NPU_HOST_LIB})

# Dynamic libraries in the stub directory can only be used for compilation
if (NOT DEFINED ENV{NPU_HOST_LIB})
    string(TOLOWER "${CMAKE_SYSTEM_NAME}" SYSTEM_NAME_LOWER)
    set(LIB_PATH "/usr/local/Ascend/ascend-toolkit/latest/${CMAKE_SYSTEM_PROCESSOR}-${SYSTEM_NAME_LOWER}/devlib")
    message(STATUS "set default LIB_PATH: ${LIB_PATH}")
else ()
    message(STATUS "env LIB_PATH: ${LIB_PATH}")
endif()


# 从环境变量读取
set(ASCEND_CUSTOM_OPP_PATH_RAW "$ENV{ASCEND_CUSTOM_OPP_PATH}")
string(REGEX REPLACE ":$" "" ASCEND_CUSTOM_OPP_PATH_CLEAN "${ASCEND_CUSTOM_OPP_PATH_RAW}")
string(REPLACE ":" ";" ASCEND_CUSTOM_OPP_PATH_LIST "${ASCEND_CUSTOM_OPP_PATH_CLEAN}")
list(GET ASCEND_CUSTOM_OPP_PATH_LIST 0 ASCEND_CUSTOM_OPP_PATH)
message(STATUS "Using ASCEND_CUSTOM_OPP_PATH: ${ASCEND_CUSTOM_OPP_PATH}")


# Header path
include_directories(
    ../inc
    ${INC_PATH}/include
    ${CUST_PKG_PATH}/include
    ${ASCEND_CUSTOM_OPP_PATH}/op_api/include

)

# add host lib path
link_directories(
    ${LIB_PATH}
    ${CUST_PKG_PATH}/lib
    ${ASCEND_CUSTOM_OPP_PATH}/op_api/lib

)

add_executable(execute_add_op
    operator_desc.cpp
    op_runner.cpp
    main.cpp
    common.cpp
)

target_link_libraries(execute_add_op
    ascendcl
    cust_opapi
    acl_op_compiler
    nnopbase
    stdc++
)

install(TARGETS execute_add_op DESTINATION ${CMAKE_RUNTIME_OUTPUT_DIRECTORY})
